package com.github.myGenerator.generator;

import com.github.myGenerator.util.ReflectUtil;

import java.io.IOException;
import java.util.List;

/**
 * Created by Cheney on 2016/8/11.
 */
public class MyGenerator {
    public static void main(String[] args) {
        MyGeneratorFacade g = new MyGeneratorFacade();
        try {
            g.getGenerator().addTemplateRootDir("template");
            //自动搜索数据库中的所有表并生成文件,template为模板的根目录
            g.generateByAllTable();
            //自定义生成某张表对应的文件
//            g.generateByTable("user_info");
            // 通过entity文件来生成对应文件，不需要数据库连接
            List<Class<?>> classes = ReflectUtil.getClasses("com.github.myGenerator.entity");
            g.generateByTable(classes);
        } catch (IOException e) {
            e.printStackTrace();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}
